# 简单封装
# 导入第三方库
import pymysql

#连接数据库
conn = pymysql.connect(host='localhost', user='root', password='123456', database='test')

# 创建游标对象
cursor = conn.cursor()

# 定义数据库操作函数
# 添加数据
def add_user(name, age):
    #具体SQL语句可根据自己实际需求编写
    sql = "INSERT INTO users(name, age) VALUES (%s, %s)"
    cursor.execute(sql, (name, age))
    #提交事务
    conn.commit()

# 修改数据
def update_user(name, age):
    #具体SQL语句可根据自己实际需求编写
    sql = "UPDATE users SET age=%s WHERE name=%s"
    cursor.execute(sql, (age, name))
    #提交事务
    conn.commit()

# 删除数据
def delete_user(name):
    #具体SQL语句可根据自己实际需求编写
    sql = "DELETE FROM users WHERE name=%s"
    cursor.execute(sql, (name,))
    #提交事务
    conn.commit()

# 查询数据
def query_user():
    #具体SQL语句可根据自己实际需求编写
    sql = "SELECT * FROM users"
    cursor.execute(sql)
    result = cursor.fetchall()
    return result

# 调用数据库操作函数
add_user('Tom', 20)
update_user('Tom', 21)
delete_user('Tom')
result = query_user()

#关闭游标和连接
cursor.close()
conn.close()